from pydantic import BaseModel
from typing import Optional

class GenerateCommentsRequest(BaseModel):
    event_id: str
    prompt_id: str
    number: int
    model_type: str
    last_editor: str

class GenerateCommentsWithNoDbRequest(BaseModel):
    prompt: str
    number: int

class GenerateCommentsWithNoDbRespond(BaseModel):
    content: str

class GenerateCommentsToDbRequest(BaseModel):
    prompt: str
    number: int
    event_id: str
    model_type: str = "Pro/deepseek-ai/DeepSeek-V3"
    last_editor: Optional[str] = None